/**
 * Description:
 * Class Name:MulticastSocketTest
 * Date:2022/5/28 14:28
 *
 * @author LTisme
 * @version JAVA17
 */
package 疯狂Java讲义.第17章_网络编程._17_4_基于UDP的网络编程._17_4_3_使用MulticastSocket实现多点广播;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.InetAddress;
import java.net.MulticastSocket;
import java.nio.charset.StandardCharsets;
import java.util.Scanner;

// 实现 Runnable 接口，这样该类的实例就可以作为线程的target
public class MulticastSocketTest implements Runnable{
    // 使用常量作为本程序的多点广播 IP 地址
    private static final String BROADCAST_IP = "230.0.0.1";
    // 使用常量作为本程序的多点广播目的地端口
    public static final int BROADCAST_PORT = 30000;
    // 定义每个数据报的大小最大为 4KB
    private static final int DATA_LEN = 4096;
    // 定义本程序的 MulticastSocket 实例
    private MulticastSocket multicastSocket;
    private InetAddress broadcastAddress;
    private Scanner scanner;
    // 定义接收网络数据的字节数组
    byte[] inBuff = new byte[DATA_LEN];
    // 以指定字节数组创建准备接收数据的DatagramPacket对象
    private DatagramPacket inPacket = new DatagramPacket(inBuff, inBuff.length);
    // 还要定义一个用于发送的DatagramPacket对象
    private DatagramPacket outPacket;

    public void init() throws IOException {
        try (
                // 创建键盘输入流
                var scan = new Scanner(System.in)
                ){
            // 创建用于发送、接收数据的 MulticastSocket 对象
            // 由于该 MulticastSocket 对象需要接收数据，所以有指定的端口
            multicastSocket = new MulticastSocket(BROADCAST_PORT);
            broadcastAddress = InetAddress.getByName(BROADCAST_IP);
            // 将该 MulticastSocket 对象加入到指定的多点广播地址
            multicastSocket.joinGroup(broadcastAddress);                            // 该方法已经陈旧了
            // 设置本 MulticastSocket 对象发送的数据报会被回送到自身
            multicastSocket.setLoopbackMode(false);                                 // 该方法已经陈旧了
            // 初始化发送用的 DatagramSocket，它包含一个长度为 0 的字节数组
            outPacket = new DatagramPacket(new byte[0], 0, broadcastAddress, BROADCAST_PORT);
            // 启动以本实例的 run() 方法作为线程执行体的线程
            new Thread(this).start();
            // 不断地读取键盘输入
            while (scan.hasNextLine()){
                // 将键盘输入的一行字符串转换成字节数组
                byte[] buff = scan.nextLine().getBytes(StandardCharsets.UTF_8);
                // 设置发送用的 DatagramPacket 里的字节数据
                outPacket.setData(buff);
                // 发送数据报
                multicastSocket.send(outPacket);
            }
        }
        finally {
            multicastSocket.close();
        }
    }

    @Override
    public void run() {
        try {
            while (true){
                // 读取 Socket 中的数据，读到的数据要放在 inPacket 所封装的字节数组里
                multicastSocket.receive(inPacket);
                // 打印输出从 多点广播中读取的内容
                System.out.println("聊天信息：" + new String(inBuff, 0, inPacket.getLength()));
            }
        }
        catch (IOException ex){
            ex.printStackTrace();
            try {
                if (multicastSocket != null){
                    // 让该 Socket 离开该多点 IP 广播地址
                    multicastSocket.leaveGroup(broadcastAddress);
                    // 关闭该 Socket 对象
                    multicastSocket.close();
                }
            }
            catch (IOException e){
                e.printStackTrace();
            }
        }
    }

    public static void main(String[] args) throws IOException{
        new MulticastSocketTest().init();
    }
}
